#pragma once
#include <string>
#include <vector>
#include <atomic>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <boost/algorithm/string.hpp>

/*
 * 工具类
 */

namespace ns_util
{
    const std::string &temp_path = "./temp/";

    class PathUtil
    {
    private:
    public:
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }
        //编译时报错的临时文件

        //通过用户上传的源文件代码，构建文件后缀+完整的路径文件名
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        //构建可执行程序的完整路径+文件后缀
        static std::string  Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        //编译是报错的文件
        static std::string CompileError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_error");
        }

        //运行时需要的临时文件

        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        // 构建该程序对应标准错误文件的完整路径+文件后缀
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };
    class TimeUtil
    {
    public:
        static std::string GetTimeStamp()
        {
            struct timeval tv;
            gettimeofday(&tv, nullptr);       //第二个参数是时区，不关心
            return std::to_string(tv.tv_sec); //秒数
        }
        //获得毫秒
        static std::string GetTimeMS()
        {
            struct timeval tv;
            gettimeofday(&tv, nullptr);
            return std::to_string(tv.tv_sec * 1000 + tv.tv_usec / 1000);
        }
    };

    class FileUtil
    {
    public:
        static bool IsFileExist(const std::string &path_name)
        {
            //获取文件属性成功，意味着文件肯定是存在的
            struct stat st;
            if (stat(path_name.c_str(), &st) == 0)
                return true;
            return false;
        }

        static std::string UniqueFileName()
        {
            //毫秒级时间戳+原子性递增唯一值：保证文件名的唯一性

            std::string ms = TimeUtil::GetTimeMS();
            static std::atomic_uint id(0);
            id++;
            std::string uq_id = std::to_string(id);
            return ms + "_" + uq_id;
        }

        //将code写入到file_name文件当中
        static bool WriteFile(const std::string &path_name, const std::string &content)
        {
            //打开文件
            std::ofstream out(path_name);
            if (!out.is_open())
                return false;

            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }

        static std::string ReadFile(const std::string &path_name, std::string *content, bool keep = false)
        {
            (*content).clear();

            std::ifstream in(path_name);
            if (!in.is_open())
                return "";

            // getline是不保留分隔符\n的，但是有时候我们是需要他保留，如何设计呢？
            std::string line;
            while (std::getline(in, line))
            {
                *content += line;
                *content += (keep ? "\n" : "");
            }
            in.close();
            return *content;
        }
    };
    class StringUtil
    {
    public:
        //将str切分存在target数组中，sep为切分字符串的分隔符
        static void StringSplit(std::string &str, std::vector<std::string> *target, const std::string &sep)
        {
            boost::split((*target), str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };

} // namespace Util
